#pragma once

#include"Common.h"

class BaseInfo
{
public:
	BaseInfo(const BaseLocation* baseLocation , Unit* hatchery , const double& distanceToMain)
		:_baseLocation(baseLocation),
		_hatchery(hatchery),
		_distanceToMain(distanceToMain)
	{
	}
	double DistanceToMain()
	{
		return _distanceToMain;
	}
	const BaseLocation* GetBaseLocation()
	{
		return _baseLocation;
	}
	Unit* GetHatchery()
	{
		return _hatchery;
	}
	void SetHatchery(Unit* hatchery)
	{
		_hatchery = hatchery;
	}
	bool operator<(const BaseInfo& another) const
	{
		return _distanceToMain < another._distanceToMain;
	}
private:
	const BaseLocation* _baseLocation;
	Unit* _hatchery;
	double _distanceToMain;
};